from gen import generate_data
from alg import max_score

def test():
    reward1, reward2, k = generate_data(n=10, k=4)
    print("reward1:", reward1)
    print("reward2:", reward2)
    print("k:", k)
    result = max_score(reward1, reward2, k)
    print("最大得分:", result)

if __name__ == "__main__":
    test()